package com.grace.controller;

import com.grace.entity.SysLog;
import com.grace.service.SysLogService;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.User;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.Date;

@Component
@Aspect
public class LogAop {


    @Autowired
    private HttpServletRequest request;
    @Autowired
    private SysLogService sysLogService;


    private Date visitTime; //开始访问时间
    private Class clazz;    //访问的类
    private Method method;  //访问的方法


    /**
     * 前置通知
     * 主要是获取开始时间，执行的类是哪一个，执行的是哪一个方法，访问的是哪一个方法
     */
    @Before("execution(* com.grace.controller.*.*(..))")
    private void doBefore(JoinPoint jp) throws NoSuchMethodException {
        visitTime = new Date();//当前时间就是开始访问的时间
        clazz = jp.getTarget().getClass();//获取当前的具体访问的类对象
        String methodName = jp.getSignature().getName();//我直接拿不到这个方法，但能获取访问方法的名称
        Object[] args = jp.getArgs();//获取访问的方法的参数

        //获取具体执行的方法的Method对象
        if (args == null || args.length == 0) {
            method = clazz.getMethod(methodName);//只能获取无参数的方法
        } else {
            Class[] classArgs = new Class[args.length]; //这里面所装的就是参数args的Class
            for (int i = 0; i < args.length; i++) {
                classArgs[i] = args[i].getClass();//把每一个参数的class拿到，装到classArgs数组里面
            }
            clazz.getMethod(methodName, classArgs);
        }
    }

    /**
     * 前置通知
     */
    @After("execution(* com.grace.controller.*.*(..))")
    private void doAfter(JoinPoint jp) {
        //获取访问时长
        long time = new Date().getTime() - visitTime.getTime();

        //获得url(classValue+methodValue)
        String url = "";
        //操作类不为空，操作方法不为空，也不能是我自己的LogAop类
        if (clazz != null && method != null && clazz != LogAop.class) {
            //1.获取类上的RequestMapping的值
            RequestMapping classAnnotation = (RequestMapping) clazz.getAnnotation(RequestMapping.class);
            if (classAnnotation != null) {
                String[] classValue = classAnnotation.value();   //class的value
                //2.获取方法上的value值
                RequestMapping methodAnnotation = method.getAnnotation(RequestMapping.class);
                if (methodAnnotation != null) {
                    String[] methodValue = methodAnnotation.value();
                    url = classValue[0] + methodValue[0];
                    //获取访问的ip地址
                    //spring 配置一个listener
                    String ip = request.getRemoteAddr();

                    //如何获取当前操作的用户
                    //方法一：从上下文中获取当前的操作用户
                    SecurityContext context = SecurityContextHolder.getContext();
                    User user = (User) context.getAuthentication().getPrincipal();
                    String username = user.getUsername();
                    //方法二：也可以通过request.getSession中获取
                    //request.getSession().getAttribute("SPRING_SECURITY_CONTEXT");


                    //将日志相关信息封装到SysLog对象
                    SysLog sysLog = new SysLog();
                    sysLog.setVisitTime(visitTime);
                    sysLog.setUsername(username);
                    sysLog.setIp(ip);
                    sysLog.setUrl(url);
                    sysLog.setExecutionTime(time);//执行时长
                    sysLog.setMethod("[类名] " + clazz.getName() + "[方法名] " + method.getName());


                    //调用Service完成插入操作
                    sysLogService.save(sysLog);
                }
            }
        }


    }
}
